package com.mgq.netty.chat.client;

import com.mgq.netty.chat.message.RpcRequestMessage;
import com.mgq.netty.chat.protocol.MessageCodecShare;
import com.mgq.netty.chat.protocol.ProtocolFrameDecoder;
import com.mgq.netty.chat.protocol.SequenceIdGenerator;
import com.mgq.netty.chat.service.HelloService;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.logging.LoggingHandler;
import io.netty.util.concurrent.DefaultPromise;
import io.netty.util.concurrent.Promise;
import lombok.extern.slf4j.Slf4j;

import java.lang.reflect.Proxy;
import java.net.InetSocketAddress;

import static com.mgq.netty.chat.client.RpcResponseHandler.PROMISE_MAP;


/**
 * 使用代理模式改造客户端
 */

@Slf4j
public class RpcClientManger {
    private static Channel channel = null;
    private static final Object LOCK = new Object();

    public static void main(String[] args) {
        Channel channel = getChannel();
        //客户端应该直接使用service接口进行调用, 不应该使用RpcRequestMessage
        //比如 客户端应该是  获取到HelloService.
        //service.sayHello("张三")
        //我们使用代理模式,当客户端使用以上方式进行调用时,我们将封装成RpcRequestMessage
        HelloService service = getProxyService(HelloService.class);
        //就是说调用接口的sayHello()方法,实际执行的是代理类的方法,代理类会进行包装,然后发送
        service.sayHello("李四");
        service.sayHello("王五");
        //接下来需要处理怎么将nettyNioEventLoop中处理的结果传给客户端,使用Promise
        //我们会有多个方法调用,会产生多个promise,为了区分,我们使用了sequenceId来进行对应

    }


    public static <T> T getProxyService(Class<T> service) {
        ClassLoader classLoader = service.getClassLoader();
        Class[] interfaces = new Class[]{service};
        Object o = Proxy.newProxyInstance(classLoader, interfaces, (proxy, method, args) -> {
            // 1. 将方法调用 转为 消息对象
            RpcRequestMessage rpcRequestMessage = new RpcRequestMessage(
                    //接口名
                    service.getName(),
                    //方法名
                    method.getName(),
                    //返回值类型
                    method.getReturnType(),
                    //参数类型
                    method.getParameterTypes(),
                    //参数
                    args
            );
            Integer id = SequenceIdGenerator.getId();
            rpcRequestMessage.setSequenceId(id);
            Channel channel = getChannel();
            // 2. 将消息对象 发送 出去               【但是 一时半会 结果不过来】
            channel.writeAndFlush(rpcRequestMessage);
            //发送完之后,调用promise.await进行等待.
            // 3. 准备一个Promise对象 来接收结果  构造方法参数【指定 Promise对象【异步】接收结果的线程】  【这里不会阻塞住，所以下面得阻塞等待结果】
            Promise<Object> promise = new DefaultPromise<>(channel.eventLoop());
            PROMISE_MAP.put(id, promise);
            // 4. await()不会抛异常， 【【【 同步阻塞 等待promise的结果(成功or失败) 】】】
            // sync()会抛出异常,我们这里不抛异常,将异常交给客户端处理
            promise.await();
            if (promise.isSuccess()) {
                //调用getNow()获取结果,返回方法的返回值
                return promise.getNow();
            } else {
                // 调用失败,将异常抛出
                throw new RuntimeException(promise.cause());
            }
        });
        return (T) o;
    }

    public static Channel getChannel() {
        if (channel != null) {
            return channel;
        }
        //双重检查
        synchronized (LOCK) {
            if (channel != null) {
                return channel;
            }
            channel = initChannel();
            return channel;
        }
    }

    private static Channel initChannel() {
        NioEventLoopGroup group = new NioEventLoopGroup();
        LoggingHandler LOG_HANDLER = new LoggingHandler();
        MessageCodecShare MES_CODEC = new MessageCodecShare();
        RpcResponseHandler RPC_RESPONSE_CODEC = new RpcResponseHandler();

        ChannelFuture future = null;
        try {
            future = new Bootstrap()
                    .group(group)
                    .option(ChannelOption.CONNECT_TIMEOUT_MILLIS, 3000)
                    .channel(NioSocketChannel.class)
                    .handler(new ChannelInitializer<NioSocketChannel>() {
                        @Override
                        protected void initChannel(NioSocketChannel ch) throws Exception {
                            ch.pipeline().addLast(new ProtocolFrameDecoder());
                            ch.pipeline().addLast(LOG_HANDLER);
                            ch.pipeline().addLast(MES_CODEC);
                            ch.pipeline().addLast(RPC_RESPONSE_CODEC);
                        }
                    }).connect(new InetSocketAddress("localhost", 8080)).sync();
        } catch (InterruptedException e) {
            log.error("error", e);
        }
        //这个位置连接已经建立
        future.channel().closeFuture().addListener(new ChannelFutureListener() {
            @Override
            public void operationComplete(ChannelFuture future) throws Exception {
                //关闭group,退出客户端
                group.shutdownGracefully();
            }
        });
        return future.channel();
    }
}
